package 买卖股票的最佳时机含手续费;

import java.util.Arrays;

/**
 * @author: AirMan
 * @date: 2025/5/16 11:14
 * @description:
 * 给定一个整数数组 prices，其中第 i 个元素代表了第 i 天的股票价格 ；非负整数 fee 代表了交易股票的手续费用。
 * 你可以无限次地完成交易，但是你每笔交易都需要付手续费。如果你已经购买了一个股票，在卖出它之前你就不能再继续购买股票了。
 * 返回获得利润的最大值。
 * 注意：这里的一笔交易指买入持有并卖出股票的整个过程，每笔交易你只需要为支付一次手续费。
 * 示例 1:
 * 输入: prices = [1, 3, 2, 8, 4, 9], fee = 2
 * 输出: 8
 * 解释: 能够达到的最大利润:
 * 在此处买入 prices[0] = 1
 * 在此处卖出 prices[3] = 8
 * 在此处买入 prices[4] = 4
 * 在此处卖出 prices[5] = 9
 * 总利润: ((8 - 1) - 2) + ((9 - 4) - 2) = 8.
 * 注意:
 * 0 < prices.length <= 50000.
 * 0 < prices[i] < 50000.
 * 0 <= fee < 50000
 */
public class Solution {
    public int maxProfit(int[] prices, int fee) {
        // 加入手续费其实就是买入的成本变高了，其他思路是一样的
        // dp数组及其下标含义：dp[i][j]表示第i天的最大金额，j一共有两种状态，持有状态和不持有状态
        // 动态转移方程：dp[i][0] = Math.max(dp[i-1][0], dp[i-1][1] - prices[i]);
        //            dp[i][1] = Math.max(dp[i-1][1], dp[i-1][0] + prices[i] - fee);
        // dp数组的初始化：dp[0][0] = -prices[0];  dp[0][1] = 0;

        // 举例推导dp数组 输入：prices = [1, 3, 2, 8, 4, 9], fee = 2

        int[][] dp = new int[prices.length][2];
        dp[0][0] = -prices[0];
        dp[0][1] = 0;
        System.out.println(Arrays.toString(dp[0]));
        for (int i = 1; i < prices.length; i++) {
            dp[i][0] = Math.max(dp[i - 1][0], dp[i - 1][1] - prices[i]);
            dp[i][1] = Math.max(dp[i - 1][0] + prices[i] - fee, dp[i - 1][1]);

            System.out.println(Arrays.toString(dp[i]));
        }

        return dp[prices.length - 1][1];
    }
}
